AI-guided patient classification